package cn.xxi.ai.util;

import org.jetbrains.annotations.NotNull;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

@Component
public class SpringContextHolder implements ApplicationContextAware {

    // 静态变量，保存 Spring 容器
    private static ApplicationContext applicationContext;

    /**
     * 实现 ApplicationContextAware 接口的回调方法。
     * 在 Spring 容器初始化时自动调用该方法注入 ApplicationContext。
     *
     * @param applicationContext Spring 应用上下文
     */
    @Override
    public void setApplicationContext(@NotNull ApplicationContext applicationContext) {
        SpringContextHolder.applicationContext = applicationContext;
    }

    /**
     * 通过 Bean 类型获取 Bean 实例。
     *
     * @param beanClass Bean 的类型
     * @param <T>       泛型，返回类型
     * @return 指定类型的 Bean 实例
     */
    public static <T> T getBean(Class<T> beanClass) {
        return applicationContext.getBean(beanClass);
    }

    /**
     * 通过 Bean 名称和类型获取 Bean 实例。
     *
     * @param beanName  Bean 的名称
     * @param beanClass Bean 的类型
     * @param <T>       泛型，返回类型
     * @return 指定名称和类型的 Bean 实例
     */
    public static <T> T getBean(String beanName, Class<T> beanClass) {
        return applicationContext.getBean(beanName, beanClass);
    }
}